package com.gzhryc.common;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Hashtable;

import javax.imageio.ImageIO;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

/**
 * 二维码图片
 */
public class MatrixToImageWriter {

	private static final int BLACK = 0xFF000000;
	private static final int WHITE = 0xFFFFFFFF;

	private MatrixToImageWriter() {}

	public static boolean StringToImgFile(String content,String path,String format,Integer width,Integer height) {
		try {
			BitMatrix bitMatrix = MatrixToImageWriter.StringToBitMatrix(content,width,height);
			 File file = new File(path);
			 if(!file.getParentFile().exists()){
				 file.mkdirs();
			 }
			 
		     MatrixToImageWriter.writeToFile(bitMatrix,format,file);
		     return true;
		} catch (IOException e) {
			e.printStackTrace();
		}
		return false;
	}

	public static BufferedImage toBufferedImage(BitMatrix matrix) {
		int width = matrix.getWidth();
		int height = matrix.getHeight();
		BufferedImage image = new BufferedImage(width, height,
				BufferedImage.TYPE_INT_RGB);
		for (int x = 0; x < width; x++) {
			for (int y = 0; y < height; y++) {
				image.setRGB(x, y, matrix.get(x, y) ? BLACK : WHITE);
			}
		}
		return image;
	}

	public static void writeToFile(BitMatrix matrix, String format, File file)
			throws IOException {
		BufferedImage image = toBufferedImage(matrix);
		if (!ImageIO.write(image, format, file)) {
			throw new IOException("Could not write an image of format "
					+ format + " to " + file);
		}
	}

	public static void writeToStream(BitMatrix matrix, String format,
			OutputStream stream) throws IOException {
		BufferedImage image = toBufferedImage(matrix);
		if (!ImageIO.write(image, format, stream)) {
			throw new IOException("Could not write an image of format "
					+ format);
		}
	}
	
	public static boolean writeToStream(String content,Integer width,Integer height,String format,OutputStream stream){
		BitMatrix matrix = MatrixToImageWriter.StringToBitMatrix(content, width, height);
		if(matrix != null){
			try {
				MatrixToImageWriter.writeToStream(matrix, format, stream);
				return true;
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		return false;
	}
	
	public static BitMatrix StringToBitMatrix(String content,Integer width,Integer height){
		try {
			MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
			Hashtable<EncodeHintType, Object> hints = new Hashtable<EncodeHintType, Object>(); 
			//指定编码格式 
			hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
			//指定纠错等级 
			hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.L); 
			//设置白边，参数值：0-4
			hints.put(EncodeHintType.MARGIN, 1); 
			BitMatrix bitMatrix = multiFormatWriter.encode(content,
					BarcodeFormat.QR_CODE, width, height, hints);
			return bitMatrix;
		} catch (WriterException e) {
			e.printStackTrace();
		}
		return null;
	}
}
